package com.ruoyi.proposal.controller;

import com.github.pagehelper.PageHelper;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.proposal.domain.Proposal;
import com.ruoyi.common.utils.PageUtils;
import com.ruoyi.proposal.dto.ProposalQueryDTO;
import com.ruoyi.proposal.service.ProposalService;
import com.ruoyi.proposal.vo.ProposalVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 提案管理API控制器
 * 对应前端manage.vue页面
 */
@RestController
@RequestMapping("/proposal")
public class ProposalManageController extends BaseController {

    @Autowired
    private ProposalService proposalService;

    /**
     * 查询提案列表
     * 对应前端manage.vue中的loadProposals方法
     * @param queryDTO 查询条件
     * @return 提案列表
     */
    @PreAuthorize("@ss.hasPermi('proposal:manage:list')")
    @GetMapping("/manage")
    public TableDataInfo listProposals(ProposalQueryDTO queryDTO) {
        // 处理月份查询条件
        if (queryDTO.getMonth() != null && !queryDTO.getMonth().isEmpty()) {
            String month = queryDTO.getMonth();
            // 如果传入的是"yyyy-MM"格式，保持原样
            if (month.matches("\\d{4}-\\d{2}")) {
                // 啥也不干
            }
            // 如果传入的是"yyyy-M"格式，补零为"yyyy-MM"
            else if (month.matches("\\d{4}-\\d{1}")) {
                String[] parts = month.split("-");
                queryDTO.setMonth(String.format("%s-0%s", parts[0], parts[1]));
            }
        }

        return proposalService.findProposalList(queryDTO);
    }

    /**
     * 查询提案详细
     * @param id 提案ID
     * @return 提案详情
     */
    @PreAuthorize("@ss.hasPermi('proposal:manage:query')")
    @GetMapping("/manage/{id}")
    public AjaxResult getProposal(@PathVariable Long id) {
        ProposalVO proposal = proposalService.findById(id);
        return AjaxResult.success(proposal);
    }

    /**
     * 新增提案
     * @param proposal 提案对象
     * @return 响应结果
     */
    @PreAuthorize("@ss.hasPermi('proposal:manage:add')")
    @Log(title = "提案管理", businessType = BusinessType.INSERT)
    @PostMapping("/manage")
    public AjaxResult addProposal(@RequestBody Proposal proposal) {
        try {
            // 调用服务新增提案
            proposalService.submitProposal(convertToDTO(proposal));
            return AjaxResult.success("新增成功");
        } catch (Exception e) {
            return AjaxResult.error("新增失败: " + e.getMessage());
        }
    }

    /**
     * 更新提案
     * 对应前端manage.vue中的updateProposal方法
     * @param proposal 提案对象
     * @return 响应结果
     */
    @PreAuthorize("@ss.hasPermi('proposal:manage:edit')")
    @Log(title = "提案管理", businessType = BusinessType.UPDATE)
    @PutMapping("/manage")
    public AjaxResult updateProposal(@RequestBody Proposal proposal) {
        try {
            // 调用服务更新提案
            proposalService.updateProposal(convertToDTO(proposal));
            return AjaxResult.success("更新成功");
        } catch (Exception e) {
            return AjaxResult.error("更新失败: " + e.getMessage());
        }
    }

    /**
     * 删除提案
     * 对应前端manage.vue中的handleDelete方法
     * @param id 提案ID
     * @return 响应结果
     */
    @PreAuthorize("@ss.hasPermi('proposal:manage:remove')")
    @Log(title = "提案管理", businessType = BusinessType.DELETE)
    @DeleteMapping("/manage/{id}")
    public AjaxResult deleteProposal(@PathVariable Long id) {
        try {
            // 调用服务删除提案
            proposalService.deleteProposal(id);
            return AjaxResult.success("删除成功");
        } catch (Exception e) {
            return AjaxResult.error("删除失败: " + e.getMessage());
        }
    }

    /**
     * 将Proposal对象转换为ProposalDTO
     * @param proposal 提案对象
     * @return 提案DTO对象
     */
    private com.ruoyi.proposal.dto.ProposalDTO convertToDTO(Proposal proposal) {
        com.ruoyi.proposal.dto.ProposalDTO dto = new com.ruoyi.proposal.dto.ProposalDTO();
        org.springframework.beans.BeanUtils.copyProperties(proposal, dto);
        return dto;
    }
}